package com.github.monkeywie.proxyee.util;

import io.netty.buffer.ByteBuf;
import io.netty.buffer.Unpooled;

import java.nio.charset.Charset;

public class ByteUtil {


    public static int findText(ByteBuf byteBuf, String str) {
        byte[] text = str.getBytes();
        for (int i = byteBuf.readerIndex(); i < byteBuf.readableBytes(); i++) {
            for (int j = 0; j < text.length; j++) {
                if (byteBuf.getByte(i+j) == text[j]) {
                    if (j == text.length-1) {
                        return i;
                    }
                } else {
                    break;
                }
            }
        }
        return -1;
    }

    public static ByteBuf insertText(ByteBuf byteBuf, int index, String str) {
        return insertText(byteBuf, index, str, Charset.defaultCharset());
    }

    public static ByteBuf insertText(ByteBuf byteBuf, int index, String str, Charset charset) {
        byte[] begin = new byte[index];
        byte[] end = new byte[byteBuf.readableBytes() - begin.length];
        byteBuf.readBytes(begin);
        byteBuf.readBytes(end);
        return Unpooled.wrappedBuffer(begin,str.getBytes(charset),end);
    }


}
